from selenium import  webdriver
import  time
from PIL import  Image
import requests
from hashlib import md5
from selenium.webdriver import ActionChains

class Chaojiying_Client(object):

    def __init__(self, username, password, soft_id):
        self.username = username
        password =  password.encode('utf8')
        self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
        }
        params.update(self.base_params)
        files = {'userfile': ('ccc.jpg', im)}
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
        return r.json()

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
        return r.json()

# 使用selenium打开界面
bro=webdriver.Chrome(executable_path='./chromedriver.exe')
bro.get('http://kyfw.12306.cn/otn/login/init')
time.sleep(1)
# 截图对打开页面进行截图并保存
bro.save_screenshot('aa.png')
# 对验证码局部区域进行裁剪确定验证码左上角和右下脚坐标x,y
code_img_ele=bro.find_element_by_xpath('//*[@id="loginForm"]/div/ul[2]/li[4]/div/div/div[3]/img')
location=code_img_ele.location#验证码左上角的坐标
size=code_img_ele.size#宽度
rangle=(
    int(location['x']),int(location['y']),int(location['x']+size['width']),int(location['y']+size['height'])
)
# 验证码图片图片区域定位下来
i=Image.open('aa.png')
code_image_name='code.png'
frame=i.crop(rangle)
frame.save(code_image_name)
# 将验证码提交给超级鹰进行识别
chaojiying=Chaojiying_Client('nidekuku', 'yunpeng123', '909415')
im=open('code.png','rb').read()
print(chaojiying.PostPic(im,9004)['pic_str'])
result=chaojiying.PostPic(im,9004)['pic_str']
all_list = []
if '|' in result:
    list_1 = result.split('|')
    count_1 = len(list_1)
    for i in range(count_1):
        xy_list = []
        x = int(list_1[i].split(',')[0])
        y = int(list_1[i].split(',')[1])
        xy_list.append(x)
        xy_list.append(y)
        all_list.append(xy_list)
else:
    x = int(result.split(',')[0])
    y = int(result.split(',')[1])
    xy_list = []
    xy_list.append(x)
    xy_list.append(y)
    all_list.append(xy_list)
print(all_list)

code_img = bro.find_element_by_xpath('//*[@id="loginForm"]/div/ul[2]/li[4]/div/div/div[3]/img')

action = ActionChains(bro)

for l in all_list:
    x = l[0]
    y = l[1]
    ActionChains(bro).move_to_element_with_offset(code_img, x, y).click().perform()

bro.find_element_by_id('username').send_keys('xxxxxxx')
time.sleep(2)
bro.find_element_by_id('password').send_keys('xxxxxx')
time.sleep(2)
bro.find_element_by_id('loginSub').click()
time.sleep(10)
bro.quit()
